Transform Weather Sensor Data from Waggle to an Xarray Dataset + Plot with ACT

Imports

import sage_data_client
from bokeh.models.formatters import DatetimeTickFormatter
import hvplot.pandas
import hvplot.xarray
import holoviews as hv
import cartopy.crs as ccrs
import cartopy.feature as cfeature
import xarray as xr
import matplotlib.pyplot as plt
from metpy.plots import USCOUNTIES
import act
import numpy as np
import pandas as pd
import warnings
from bokeh.models import DatetimeTickFormatter

def apply_formatter(plot, element):
    plot.handles['xaxis'].formatter = DatetimeTickFormatter(hours='%m/%d/%Y \n %H:%M',
                                                            minutes='%m/%d/%Y \n %H:%M',
                                                            hourmin='%m/%d/%Y \n %H:%M',
                                                            days='%m/%d/%Y \n %H:%M',
                                                            months='%m/%d/%Y \n %H:%M')
xr.set_options(keep_attrs=True)
warnings.filterwarnings("ignore")
hv.extension("bokeh")

Query for the available Data

wxt_df = sage_data_client.query(
    start="-3h",
    filter={
        "sensor": "vaisala-wxt536"
    }
)

Configure Helper Functions and Renaming Conventions

The renaming is required due to . notations being problematic when working with both Pandas and Xarray data structures.

variable_rename_dict = {'wxt.env.humidity':'air_humidity',
                        'wxt.env.pressure':'air_pressure',
                        'wxt.env.temp':'air_temperature',
                        'wxt.heater.temp':'heater_temperature',
                        'wxt.heater.volt':'heater_voltage',
                        'wxt.rain.accumulation':'rain_accumulation',
                        'wxt.wind.direction':'wind_direction',
                        'wxt.wind.speed':'wind_speed',
                        'sys.gps.lat':'latitude',
                        'sys.gps.lon':'longitude',
                    }

def generate_data_array(df, variable, rename_variable_dict=variable_rename_dict):
    new_variable_name = rename_variable_dict[variable]
    df_variable= df.loc[df.name == variable]
    ds = df_variable.to_xarray().rename({'value':new_variable_name,
                                         'timestamp':'time',
                                         'meta.vsn':'node'})
    ds[new_variable_name].attrs['units'] = df_variable['meta.units'].values[0]
    ds['time'] = pd.to_datetime(ds.time)
    ds.attrs['datastream'] = ds.node.values[0]
    return ds[[new_variable_name]]

def generate_dataset(df, variables, rename_variable_dict=variable_rename_dict):
    reindexed = df.set_index(['meta.vsn', 'timestamp'])
    return xr.merge([generate_data_array(reindexed, variable) for variable in variables])

Transform the Data to Xarray

wxt_variables = wxt_df.name.unique()
wxt_variables

array(['wxt.env.humidity', 'wxt.env.pressure', 'wxt.env.temp',
       'wxt.heater.temp', 'wxt.heater.volt', 'wxt.rain.accumulation',
       'wxt.wind.direction', 'wxt.wind.speed'], dtype=object)

wxt_ds = generate_dataset(wxt_df, wxt_variables).squeeze()
wxt_ds

<xarray.Dataset> Size: 82MB
Dimensions:             (node: 4, time: 311702)
Coordinates:
  * node                (node) object 32B 'W01B' 'W067' 'W08D' 'W099'
  * time                (time) datetime64[ns] 2MB 2024-04-09T11:45:16.5387165...
Data variables:
    air_humidity        (node, time) float64 10MB nan 67.6 nan ... nan nan nan
    air_pressure        (node, time) float64 10MB nan 997.8 nan ... nan 988.6
    air_temperature     (node, time) float64 10MB nan 9.7 nan ... 15.7 nan 15.7
    heater_temperature  (node, time) float64 10MB nan 13.4 nan ... 19.6 nan 19.6
    heater_voltage      (node, time) float64 10MB nan 11.4 nan ... 23.9 nan 23.8
    rain_accumulation   (node, time) float64 10MB nan 342.1 nan ... nan 311.0
    wind_direction      (node, time) float64 10MB nan 220.0 nan ... nan 261.0
    wind_speed          (node, time) float64 10MB nan 1.6 nan ... 2.3 nan 2.3
Attributes:
    datastream:  W01B

xarray.Dataset

• Dimensions:
  • node: 4
  • time: 311702
• Coordinates: (2)
  • node
    (node)
    object
    'W01B' 'W067' 'W08D' 'W099'

    array(['W01B', 'W067', 'W08D', 'W099'], dtype=object)

  • time
    (time)
    datetime64[ns]
    2024-04-09T11:45:16.538716596 .....

    array(['2024-04-09T11:45:16.538716596', '2024-04-09T11:45:16.595380453',
           '2024-04-09T11:45:16.618497816', ..., '2024-04-09T14:45:16.477050963',
           '2024-04-09T14:45:16.546811716', '2024-04-09T14:45:16.557052890'],
          dtype='datetime64[ns]')

• Data variables: (8)
  • air_humidity
    (node, time)
    float64
    nan 67.6 nan 67.6 ... nan nan nan

    units :

    percent

    array([[ nan, 67.6,  nan, ...,  nan,  nan,  nan],
           [ nan,  nan,  nan, ...,  nan,  nan,  nan],
           [ nan,  nan,  nan, ...,  nan,  nan,  nan],
           [61.4,  nan, 61.4, ...,  nan,  nan,  nan]])

  • air_pressure
    (node, time)
    float64
    nan 997.8 nan ... 988.6 nan 988.6

    units :

    hectoPascal

    array([[  nan, 997.8,   nan, ...,   nan, 998.8,   nan],
           [  nan,   nan,   nan, ...,   nan,   nan,   nan],
           [  nan,   nan,   nan, ...,   nan,   nan,   nan],
           [987.4,   nan, 987.4, ..., 988.6,   nan, 988.6]])

  • air_temperature
    (node, time)
    float64
    nan 9.7 nan 9.7 ... 15.7 nan 15.7

    units :

    degree Celsius

    array([[ nan,  9.7,  nan, ...,  nan, 11.9,  nan],
           [ nan,  nan,  nan, ...,  nan,  nan,  nan],
           [ nan,  nan,  nan, ...,  nan,  nan,  nan],
           [12.5,  nan, 12.5, ..., 15.7,  nan, 15.7]])

  • heater_temperature
    (node, time)
    float64
    nan 13.4 nan 13.4 ... 19.6 nan 19.6

    units :

    degree Celsius

    array([[ nan, 13.4,  nan, ...,  nan, 15.2,  nan],
           [ nan,  nan,  nan, ...,  nan,  nan,  nan],
           [ nan,  nan,  nan, ...,  nan,  nan,  nan],
           [15. ,  nan, 15. , ..., 19.6,  nan, 19.6]])

  • heater_voltage
    (node, time)
    float64
    nan 11.4 nan 11.4 ... 23.9 nan 23.8

    units :

    volts

    array([[ nan, 11.4,  nan, ...,  nan, 12.1,  nan],
           [ nan,  nan,  nan, ...,  nan,  nan,  nan],
           [ nan,  nan,  nan, ...,  nan,  nan,  nan],
           [23.9,  nan, 23.9, ..., 23.9,  nan, 23.8]])

  • rain_accumulation
    (node, time)
    float64
    nan 342.1 nan ... 311.0 nan 311.0

    units :

    milimeters

    array([[   nan, 342.13,    nan, ...,    nan, 342.13,    nan],
           [   nan,    nan,    nan, ...,    nan,    nan,    nan],
           [   nan,    nan,    nan, ...,    nan,    nan,    nan],
           [310.98,    nan, 310.98, ..., 310.98,    nan, 310.98]])

  • wind_direction
    (node, time)
    float64
    nan 220.0 nan ... 261.0 nan 261.0

    units :

    degrees

    array([[ nan, 220.,  nan, ...,  nan, 241.,  nan],
           [ nan,  nan,  nan, ...,  nan,  nan,  nan],
           [ nan,  nan,  nan, ...,  nan,  nan,  nan],
           [237.,  nan, 237., ..., 261.,  nan, 261.]])

  • wind_speed
    (node, time)
    float64
    nan 1.6 nan 1.6 ... nan 2.3 nan 2.3

    units :

    meters per second

    array([[nan, 1.6, nan, ..., nan, 2.1, nan],
           [nan, nan, nan, ..., nan, nan, nan],
           [nan, nan, nan, ..., nan, nan, nan],
           [1.9, nan, 1.9, ..., 2.3, nan, 2.3]])

• Indexes: (2)
  • node
    PandasIndex

    PandasIndex(Index(['W01B', 'W067', 'W08D', 'W099'], dtype='object', name='node'))

  • time
    PandasIndex

    PandasIndex(DatetimeIndex(['2024-04-09 11:45:16.538716596',
                   '2024-04-09 11:45:16.595380453',
                   '2024-04-09 11:45:16.618497816',
                   '2024-04-09 11:45:16.671656536',
                   '2024-04-09 11:45:16.699047103',
                   '2024-04-09 11:45:16.747040959',
                   '2024-04-09 11:45:16.778142145',
                   '2024-04-09 11:45:16.787912784',
                   '2024-04-09 11:45:16.827801777',
                   '2024-04-09 11:45:16.858367494',
                   ...
                   '2024-04-09 14:45:16.217112329',
                   '2024-04-09 14:45:16.237104350',
                   '2024-04-09 14:45:16.312878462',
                   '2024-04-09 14:45:16.317911946',
                   '2024-04-09 14:45:16.388440278',
                   '2024-04-09 14:45:16.397120940',
                   '2024-04-09 14:45:16.463243548',
                   '2024-04-09 14:45:16.477050963',
                   '2024-04-09 14:45:16.546811716',
                   '2024-04-09 14:45:16.557052890'],
                  dtype='datetime64[ns]', name='time', length=311702, freq=None))

• Attributes: (1)

  datastream :

  W01B

Resample the data to minute requency

minute_ds = wxt_ds.resample(time='1T').mean()

Visualize using hvplot

meteogram_variables = ['air_temperature', 'air_humidity', 'wind_speed', 'wind_direction']
plots = []
for variable in meteogram_variables:
    plots.append(wxt_ds[variable].hvplot.line(label='10 Hz Data') * 
                 minute_ds[variable].hvplot.line(label='1 Minute Data'))
hv.Layout(plots).cols(2)

Plot Using the Atmospheric Data Community Toolkit (ACT)

WindDisplay = act.plotting.WindRoseDisplay(minute_ds, figsize=(8, 10), subplot_shape=(1,))
WindDisplay.plot(
    'wind_direction', 'wind_speed', spd_bins=np.linspace(0, 25, 5), num_dirs=30, tick_interval=2, subplot_index=(0,)
)

<PolarAxes: title={'center': 'W01B on 20240409'}>